# shellcheck shell=bash
# shellcheck disable=SC2120,SC2207

cite "about-completion"
about-completion "virtualbox - Oracle VM VirtualBox virtualization software"
group "virtualization"
url "https://www.virtualbox.org/"

_vboxmanage_realopts() {
	vboxmanage | grep -i vboxmanage | cut -d' ' -f2 | grep '\[' | tr -s '\[\|\]\n' ' '
	echo " "
}

__vboxmanage_startvm() {
	RUNNING=$(vboxmanage list runningvms | cut -d' ' -f1 | tr -d '"')
	TOTAL=$(vboxmanage list vms | cut -d' ' -f1 | tr -d '"')

	AVAILABLE=""
	for VM in $TOTAL; do
		MATCH=0
		for RUN in $RUNNING "x"; do
			if [ "$VM" == "$RUN" ]; then
				MATCH=1
			fi
		done
		((MATCH == 0)) && AVAILABLE="$AVAILABLE $VM "
	done
	echo "$AVAILABLE"
}

__vboxmanage_list() {
	INPUT=$(vboxmanage list | tr -s '\[\]\|\n' ' ' | cut -d' ' -f4-)

	PRUNED=""
	if [ "$1" == "long" ]; then
		for WORD in $INPUT; do
			[ "$WORD" == "-l" ] && continue
			[ "$WORD" == "--long" ] && continue

			PRUNED="$PRUNED $WORD"
		done
	else
		PRUNED=$INPUT
	fi

	echo "$PRUNED"
}

__vboxmanage_list_vms() {
	VMS=""
	if [ "$1" == "" ]; then
		SEPARATOR=" "
	else
		SEPARATOR=$1
	fi

	for VM in $(vboxmanage list vms | cut -d' ' -f1 | tr -d '"'); do
		[ "$VMS" != "" ] && VMS="${VMS}${SEPARATOR}"
		VMS="${VMS}${VM}"
	done

	echo "$VMS"
}

__vboxmanage_list_runningvms() {
	VMS=""
	if [ "$1" == "" ]; then
		SEPARATOR=" "
	else
		SEPARATOR=$1
	fi

	for VM in $(vboxmanage list runningvms | cut -d' ' -f1 | tr -d '"'); do
		[ "$VMS" != "" ] && VMS="${VMS}${SEPARATOR}"
		VMS="${VMS}${VM}"
	done

	echo "$VMS"

}

__vboxmanage_controlvm() {
	echo "pause resume reset poweroff savestate acpipowerbutton"
	echo "acpisleepbutton keyboardputscancode guestmemoryballoon"
	echo "gueststatisticsinterval usbattach usbdetach vrde vrdeport"
	echo "vrdeproperty vrdevideochannelquality setvideomodehint"
	echo "screenshotpng setcredentials teleport plugcpu unplugcpu"
	echo "cpuexecutioncap"

	# setlinkstate<1-N>
	# nic<1-N> null|nat|bridged|intnet|hostonly|generic
	#                                      [<devicename>] |
	# nictrace<1-N> on|off
	#   nictracefile<1-N> <filename>
	#   nicproperty<1-N> name=[value]
	#   natpf<1-N> [<rulename>],tcp|udp,[<hostip>],
	#                 <hostport>,[<guestip>],<guestport>
	#   natpf<1-N> delete <rulename>

}

__vboxmanage_default() {
	realopts=$(_vboxmanage_realopts)
	opts=$realopts$(vboxmanage | grep -i vboxmanage | cut -d' ' -f2 | grep -v '\[' | sort | uniq)
	pruned=""

	# echo ""
	# echo "DEBUG: cur: $cur, prev: $prev"
	# echo "DEBUG: default: |$p1|$p2|$p3|$p4|"
	case ${cur} in
		-*)
			echo "$opts"
			# COMPREPLY=($(compgen -W "${opts}" -- ${cur}))
			return 0
			;;
	esac

	for WORD in $opts; do
		MATCH=0
		for OPT in "${COMP_WORDS[@]}"; do
			# opts=$(echo "${opts}" | grep -v $OPT);
			if [ "$OPT" == "$WORD" ]; then
				MATCH=1
				break
			fi
			if [ "$OPT" == "-v" ] && [ "$WORD" == "--version" ]; then
				MATCH=1
				break
			fi
			if [ "$OPT" == "--version" ] && [ "$WORD" == "-v" ]; then
				MATCH=1
				break
			fi
			if [ "$OPT" == "-q" ] && [ "$WORD" == "--nologo" ]; then
				MATCH=1
				break
			fi
			if [ "$OPT" == "--nologo" ] && [ "$WORD" == "-q" ]; then
				MATCH=1
				break
			fi
		done
		((MATCH == 1)) && continue
		pruned="$pruned $WORD"

	done

	# COMPREPLY=($(compgen -W "${pruned}" -- "${cur}"))
	echo "$pruned"
	return 0
}

_vboxmanage() {
	# vboxmanage | grep -i vboxmanage | cut -d' ' -f2 | sort | uniq
	local cur opts
	COMPREPLY=()
	cur="${COMP_WORDS[COMP_CWORD]}"
	prev="${COMP_WORDS[COMP_CWORD - 1]}"

	# echo "cur: |$cur|"
	# echo "prev: |$prev|"

	# In case current is complete command
	case $cur in
		startvm | list | controlvm)
			COMPREPLY=($(compgen -W "${cur}"))
			return 0
			;;
	esac

	case $prev in
		-v | --version)
			return 0
			;;

		-l | --long)
			opts=$(__vboxmanage_list "long")
			COMPREPLY=($(compgen -W "${opts}" -- "${cur}"))
			return 0
			;;
		startvm | list)
			opts=$(__vboxmanage_"$prev")
			COMPREPLY=($(compgen -W "${opts}" -- "${cur}"))
			return 0
			;;
		--type)
			COMPREPLY=($(compgen -W "gui headless" -- "${cur}"))
			return 0
			;;
		gui | headless)
			# Done. no more completion possible
			return 0
			;;
		vboxmanage | -q | --nologo)
			# echo "Got vboxmanage"
			opts=$(__vboxmanage_default)
			COMPREPLY=($(compgen -W "${opts}" -- "${cur}"))
			return 0
			;;
		controlvm)
			opts=$(__vboxmanage_list_vms)
			COMPREPLY=($(compgen -W "${opts}" -- "${cur}"))
			return 0
			;;
	esac

	for VM in $(__vboxmanage_list_vms); do
		if [ "$VM" == "$prev" ]; then
			pprev=${COMP_WORDS[COMP_CWORD - 2]}
			# echo "previous: $pprev"
			case $pprev in
				startvm)
					opts="--type"
					COMPREPLY=($(compgen -W "${opts}" -- "${cur}"))
					return 0
					;;
				controlvm)
					opts=$(__vboxmanage_controlvm)
					COMPREPLY=($(compgen -W "${opts}" -- "${cur}"))
					return 0
					;;
			esac
		fi
	done

	# echo "Got to end withoug completion"
}
complete -F _vboxmanage vboxmanage
